package mine.code.question.动态规划;

import org.junit.Test;

import java.util.Arrays;

/**
 * 在计算机界中，我们总是追求用有限的资源获取最大的收益。
 * <p>
 * 现在，假设你分别支配着 m 个 0 和 n 个 1。另外，还有一个仅包含 0 和 1 字符串的数组。
 * <p>
 * 你的任务是使用给定的 m 个 0 和 n 个 1 ，找到能拼出存在于数组中的字符串的最大数量。每个 0 和 1 至多被使用一次。
 * <p>
 * 注意:
 * <p>
 * 给定 0 和 1 的数量都不会超过 100。
 * 给定字符串数组的长度不会超过 600。
 * 示例
 * <p>
 * 输入: Array = {"10", "0", "1"}, m = 1, n = 1
 * 输出: 2
 * <p>
 * 解释: 你可以拼出 "10"，但之后就没有剩余数字了。更好的选择是拼出 "0" 和 "1" 。
 *
 * @author caijinnan
 * @date 2020/4/8 16:04
 */
public class 一和零 {

    @Test
    public void run() {
        //["111","1000","1000","1000"]
        //9
        //3
        String[] strs = {"10", "0001", "111001", "1", "0"};
        int m = 1;
        int n = 1;
        System.out.println(findMaxForm(strs, m, n));
    }

    // dp[i][j] = max( 1+ dp[i-const0[k]][j-const1[k]]
    public int findMaxForm(String[] strs, int m, int n) {
        int[][] dp = new int[m + 1][n + 1];
        dp[0][0] = 0;
        int maxNum = 0;
        for (String str : strs) {
            int zeroNum = getZero(str);
            int oneNum = str.length() - zeroNum;
            for (int i = m; i >= zeroNum; i--) {
                for (int j = n; j >= oneNum; j--) {
                    int x = i - zeroNum;
                    int y = j - oneNum;
                    dp[i][j] = Math.max(1 + dp[x][y], dp[i][j]);
                    maxNum = Math.max(maxNum, dp[i][j]);
                }
            }
        }

        return maxNum;
    }

    private int getZero(String str) {
        int num = 0;
        for (char c : str.toCharArray()) {
            if (c == '0') {
                num++;
            }
        }
        return num;
    }
}